package developer.算法.栈.最小栈;

import java.util.*;

/**
 * @author zhangyongkang
 * @time 2025/4/2 17:38
 * @description 设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。
 * <p>
 * 实现 MinStack 类:
 * <p>
 * MinStack() 初始化堆栈对象。
 * void push(int val) 将元素val推入堆栈。
 * void pop() 删除堆栈顶部的元素。
 * int top() 获取堆栈顶部的元素。
 * int getMin() 获取堆栈中的最小元素。
 * <p>
 * <p>
 * 示例 1:
 * <p>
 * 输入：
 * ["MinStack","push","push","push","getMin","pop","top","getMin"]
 * [[],[-2],[0],[-3],[],[],[],[]]
 * <p>
 * 输出：
 * [null,null,null,null,-3,null,0,-2]
 * <p>
 * 解释：
 * MinStack minStack = new MinStack();
 * minStack.push(-2);
 * minStack.push(0);
 * minStack.push(-3);
 * minStack.getMin();   --> 返回 -3.
 * minStack.pop();
 * minStack.top();      --> 返回 0.
 * minStack.getMin();   --> 返回 -2.
 * <p>
 * <p>
 * 提示：
 * <p>
 * -231 <= val <= 231 - 1
 * pop、top 和 getMin 操作总是在 非空栈 上调用
 * push, pop, top, and getMin最多被调用 3 * 104 次
 */
public class MinStackDemo {

    public static void main(String[] args) {
        MinStack minStack = new MinStack();
        minStack.push(-2);
        minStack.push(0);
        minStack.push(-3);
        System.out.println(minStack.getMin());  // --> 返回 -3.
        minStack.pop();
        System.out.println(minStack.top());     // --> 返回 0.
        System.out.println(minStack.getMin());  // --> 返回 -2.
    }

    /**
     * Your MinStackDemo object will be instantiated and called as such:
     * MinStackDemo obj = new MinStackDemo();
     * obj.push(val);
     * obj.pop();
     * int param_3 = obj.top();
     * int param_4 = obj.getMin();
     */
    static class MinStack {

        Stack<Integer> stack;
        Map<Integer, Integer> treeMap;

        public MinStack() {
            stack = new Stack<>();
            treeMap = new TreeMap<>();
        }

        public void push(int val) {
            stack.push(val);
            Integer nums = treeMap.getOrDefault(val, 0);
            treeMap.put(val, nums + 1);
        }

        public void pop() {
            Integer pop = stack.pop();
            Integer times = treeMap.getOrDefault(pop, 0);
            int nowTimes = times - 1;
            if (nowTimes <= 0) {
                treeMap.remove(pop);
            } else {
                treeMap.put(pop, nowTimes);
            }


        }

        public int top() {
            return stack.peek();
        }

        public int getMin() {
            return treeMap.keySet().stream().findFirst().get();
        }
    }

    /**
     *
     作者：力扣官方题解
     链接：https://leetcode.cn/problems/min-stack/solutions/242190/zui-xiao-zhan-by-leetcode-solution/
     来源：力扣（LeetCode）
     著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
     */
    class MinStackOfficial {
        Deque<Integer> xStack;
        Deque<Integer> minStack;

        public MinStackOfficial() {
            xStack = new LinkedList<Integer>();
            minStack = new LinkedList<Integer>();
            minStack.push(Integer.MAX_VALUE);
        }

        public void push(int x) {
            xStack.push(x);
            minStack.push(Math.min(minStack.peek(), x));
        }

        public void pop() {
            xStack.pop();
            minStack.pop();
        }

        public int top() {
            return xStack.peek();
        }

        public int getMin() {
            return minStack.peek();
        }
    }


}
